package com.itheima.reggie.controller;

import com.itheima.reggie.common.R;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.util.UUID;

/**
 * @author ：石破天
 * @date ：Created in 2022年11月20日
 * @description ：
 * @version: 1.0
 */
@Slf4j
@RestController
@RequestMapping("/common")
public class CommonController {

    @Value("${reggie.path}")
    private  String basePath ;

    //上传功能
    @PostMapping("/upload")
    public R<String> upload(MultipartFile file){

        // 是不是 后缀名得对应...
        // 后缀名是不是在文件名上呢  .后面就是后缀名
        // 需要获取上传的哪个文件的后缀名
        String originalFilename = file.getOriginalFilename();//获取原始的文件名字
        //   a.png
       log.info("上传文件的原始名字:"+originalFilename);
        // 截取后缀名  获取 . 位置的索引  从.位置往后截取
        int lastIndex = originalFilename.lastIndexOf(".");
        // 从.位置往后截取
        String suffix = originalFilename.substring(lastIndex);
        log.info("查看后缀名:"+suffix);
        // 文件名字尽量去变得不重复
        //  UUID 生成一个永不重复的字符串....
        String fileName = UUID.randomUUID().toString()+suffix;
                          // 随机名字 + 后缀名  形成新名字
        log.info("新的文件名字:"+fileName);
       // 目的地目录    文件要上传到目的地目录下...


        // 先封装 去判断  这个路径存不存在
        File dir = new File(basePath);
        if(!dir.exists()){//判断路径是否真实存在的
            // 如果不存在 创建目录
            dir.mkdirs();//如果目录不存在就创建出来
        }
        // 创建目的地文件
        File destFile = new File(dir,fileName);
                      // 在指定目录下 搞一个 叫 fileName的文件

        try {
            file.transferTo(destFile);
        } catch (IOException e) {
            e.printStackTrace();
            return R.error("上传失败");
        }

        return R.success(fileName);
    }


    //下载功能
    @GetMapping("/download")
    public void download(String name, HttpServletResponse response){

       try {
           //1:根据name名字 去目录下找到对应的文件 封装成file对象
           File file = new File(basePath, name);
           //  file就是被下载的文件
           //2: 先把被下载文件资源读到内存
           //  创建字节输入流 关联下载的文件  下载的文件就是源文件
           FileInputStream in = new FileInputStream(file);
           //3: 目的地 下载到浏览器上   response.getOutputStream() 内存通向浏览器的流
           ServletOutputStream out = response.getOutputStream();
           // 4: 边读边读
           int len = 0;
           byte[] buffer = new byte[1024*8];
           while((len=in.read(buffer))!=-1){
               // 写出去
               out.write(buffer,0,len);
           }
           //释放资源
           in.close();


       }catch (Exception e){
           System.out.println("程序下载失败");
           e.printStackTrace();
       }
    }
}
